/*
 * Copyright © 2016-2025 The LmdbJava Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.lmdbjava;

import static org.assertj.core.api.Assertions.assertThat;

import java.time.Duration;
import java.time.Instant;
import java.util.Arrays;
import java.util.Collection;
import java.util.EnumSet;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.IntStream;
import org.assertj.core.api.Assertions;
import org.junit.jupiter.api.Test;

class PutFlagSetTest extends AbstractFlagSetTest<PutFlags, PutFlagSet> {

  @Test
  void test() {
    // This is here purely to stop CodeQL moaning that this class is unused.
    // All the actual tests are in the superclass
    Assertions.assertThat(getAllFlags()).isNotNull();
  }

  @Override
  List<PutFlags> getAllFlags() {
    return Arrays.stream(PutFlags.values()).collect(Collectors.toList());
  }

  @Override
  PutFlagSet getEmptyFlagSet() {
    return PutFlagSet.empty();
  }

  @Override
  AbstractFlagSet.Builder<PutFlags, PutFlagSet> getBuilder() {
    return PutFlagSet.builder();
  }

  @Override
  PutFlagSet getFlagSet(Collection<PutFlags> flags) {
    return PutFlagSet.of(flags);
  }

  @Override
  PutFlagSet getFlagSet(PutFlags[] flags) {
    return PutFlagSet.of(flags);
  }

  @Override
  PutFlagSet getFlagSet(PutFlags flag) {
    return PutFlagSet.of(flag);
  }

  @Override
  Class<PutFlags> getFlagType() {
    return PutFlags.class;
  }

  @Override
  Function<EnumSet<PutFlags>, PutFlagSet> getConstructor() {
    return PutFlagSetImpl::new;
  }

  /**
   * {@link FlagSet#isSet(MaskedFlag)} on the flag enum is tested in {@link AbstractFlagSetTest} but
   * the coverage check doesn't seem to notice it.
   */
  @Test
  void testIsSet() {
    assertThat(PutFlags.MDB_APPEND.isSet(PutFlags.MDB_APPEND)).isTrue();
    assertThat(PutFlags.MDB_APPEND.isSet(PutFlags.MDB_MULTIPLE)).isFalse();
    //noinspection ConstantValue
    assertThat(PutFlags.MDB_APPEND.isSet(null)).isFalse();
  }

  @Test
  public void testAddFlagVsCheckPresence() {

    final int cnt = 10_000_000;
    final int[] arr = new int[cnt];
    final List<PutFlagSet> flagSets =
        IntStream.range(0, cnt)
            .boxed()
            .map(
                i ->
                    PutFlagSet.of(
                        PutFlags.MDB_APPEND, PutFlags.MDB_NOOVERWRITE, PutFlags.MDB_RESERVE))
            .collect(Collectors.toList());

    Instant time;
    for (int i = 0; i < 5; i++) {
      time = Instant.now();
      for (int j = 0; j < flagSets.size(); j++) {
        PutFlagSet flagSet = flagSets.get(j);
        if (!flagSet.isSet(PutFlags.MDB_RESERVE)) {
          throw new RuntimeException("Not set");
        }
        arr[j] = flagSet.getMask();
      }
      System.out.println("Check: " + Duration.between(time, Instant.now()));

      time = Instant.now();
      for (int j = 0; j < flagSets.size(); j++) {
        PutFlagSet flagSet = flagSets.get(j);
        final int mask = flagSet.getMaskWith(PutFlags.MDB_RESERVE);
        arr[j] = mask;
      }
      System.out.println("Append:" + Duration.between(time, Instant.now()));
    }
  }
}
